package org.linlinjava.litemall.db.util;


import java.awt.Color;
import java.awt.Graphics2D;
import java.awt.Image;
import java.awt.Rectangle;
import java.awt.image.BufferedImage;
import java.awt.image.ColorModel;
import java.io.*;
import java.util.Iterator;
import javax.imageio.*;
import javax.imageio.stream.ImageInputStream;


/**
 * 图片压缩工具类 提供的方法中可以设定生成的 缩略图片的大小尺寸、压缩尺寸的比例、图片的质量等
 * <pre>
 * 	调用示例：
 * resiz(srcImg, tarDir + "car_1_maxLength_11-220px-hui.jpg", 220, 0.7F);
 * </pre>
 */
public class ImageUtil {

    /**
     * * 图片文件读取
     *
     * @param srcImgPath
     * @return
     */
    private static BufferedImage InputImage(String srcImgPath) throws RuntimeException {

        BufferedImage srcImage = null;
        FileInputStream in = null;
        try {
            // 构造BufferedImage对象
            File file = new File(srcImgPath);
            in = new FileInputStream(file);
            byte[] b = new byte[5];
            in.read(b);
            srcImage = javax.imageio.ImageIO.read(file);
        } catch (IOException e) {
            e.printStackTrace();
            throw new RuntimeException("读取图片文件出错！", e);
        } finally {
            if (in != null) {
                try {
                    in.close();
                } catch (IOException e) {
                    throw new RuntimeException("读取图片文件出错！", e);
                }
            }
        }
        return srcImage;
    }

    /**
     * * 将图片按照指定的图片尺寸、源图片质量压缩(默认质量为1)
     *
     * @param srcImgPath
     *            :源图片路径
     * @param outImgPath
     *            :输出的压缩图片的路径
     * @param new_w
     *            :压缩后的图片宽
     * @param new_h
     *            :压缩后的图片高
     */
    public static void resize(String srcImgPath, String outImgPath,
                              int new_w, int new_h) {
        resize(srcImgPath, outImgPath, new_w, new_h, 1F);
    }

    /**
     * 将图片按照指定的尺寸比例、源图片质量压缩(默认质量为1)
     *
     * @param srcImgPath
     *            :源图片路径
     * @param outImgPath
     *            :输出的压缩图片的路径
     * @param ratio
     *            :压缩后的图片尺寸比例
     */
    public static void resize(String srcImgPath, String outImgPath,
                              float ratio) {
        resize(srcImgPath, outImgPath, ratio, 1F);
    }

    /**
     * 将图片按照指定长或者宽的最大值来压缩图片(默认质量为1)
     *
     * @param srcImgPath
     *            :源图片路径
     * @param outImgPath
     *            :输出的压缩图片的路径
     * @param maxLength
     *            :长或者宽的最大值
     */
    public static void resize(String srcImgPath, String outImgPath,
                              int maxLength) {
        resize(srcImgPath, outImgPath, maxLength, 1F);
    }

    /**
     * * 将图片按照指定的图片尺寸、图片质量压缩
     *
     * @param srcImgPath
     *            :源图片路径
     * @param outImgPath
     *            :输出的压缩图片的路径
     * @param new_w
     *            :压缩后的图片宽
     * @param new_h
     *            :压缩后的图片高
     * @param per
     *            :百分比
     * @author cevencheng
     */
    public static void resize(String srcImgPath, String outImgPath,
                              int new_w, int new_h, float per) {
        // 得到图片
        BufferedImage src = InputImage(srcImgPath);
        int old_w = src.getWidth();
        // 得到源图宽
        int old_h = src.getHeight();
        // 得到源图长
        // 根据原图的大小生成空白画布
        BufferedImage tempImg = new BufferedImage(old_w, old_h,
                BufferedImage.TYPE_INT_RGB);
        // 在新的画布上生成原图的缩略图
        Graphics2D g = tempImg.createGraphics();
        g.setColor(Color.white);
        g.fillRect(0, 0, old_w, old_h);
        g.drawImage(src, 0, 0, old_w, old_h, Color.white, null);
        g.dispose();
        BufferedImage newImg = new BufferedImage(new_w, new_h,
                BufferedImage.TYPE_INT_RGB);
        newImg.getGraphics().drawImage(
                tempImg.getScaledInstance(new_w, new_h, Image.SCALE_SMOOTH), 0,
                0, null);
        // 调用方法输出图片文件
        outImage(outImgPath, newImg, per);
    }

    /**
     * * 将图片按照指定的尺寸比例、图片质量压缩
     *
     * @param srcImgPath
     *            :源图片路径
     * @param outImgPath
     *            :输出的压缩图片的路径
     * @param ratio
     *            :压缩后的图片尺寸比例
     * @param per
     *            :百分比
     * @author cevencheng
     */
    public static void resize(String srcImgPath, String outImgPath,
                              float ratio, float per) {
        // 得到图片
        BufferedImage src = InputImage(srcImgPath);
        int old_w = src.getWidth();
        // 得到源图宽
        int old_h = src.getHeight();
        // 得到源图长
        int new_w = 0;
        // 新图的宽
        int new_h = 0;
        // 新图的长
        BufferedImage tempImg = new BufferedImage(old_w, old_h,
                BufferedImage.TYPE_INT_RGB);
        Graphics2D g = tempImg.createGraphics();
        g.setColor(Color.white);
        // 从原图上取颜色绘制新图g.fillRect(0, 0, old_w, old_h);
        g.drawImage(src, 0, 0, old_w, old_h, Color.white, null);
        g.dispose();
        // 根据图片尺寸压缩比得到新图的尺寸new_w = (int) Math.round(old_w * ratio);
        new_h =  Math.round(old_h * ratio);
        new_w =  Math.round(old_w * ratio);
        BufferedImage newImg = new BufferedImage(new_w, new_h,
                BufferedImage.TYPE_INT_RGB);
        newImg.getGraphics().drawImage(
                tempImg.getScaledInstance(new_w, new_h, Image.SCALE_SMOOTH), 0,
                0, null);
        // 调用方法输出图片文件OutImage(outImgPath, newImg, per);
        outImage(outImgPath, newImg, per);
    }

    /**
     * <b>
     * 指定长或者宽的最大值来压缩图片
     * 	推荐使用此方法
     * </b>
     * @param srcImgPath
     *            :源图片路径
     * @param outImgPath
     *            :输出的压缩图片的路径
     * @param maxLength
     *            :长或者宽的最大值
     * @param per
     *            :图片质量
     * @author cevencheng
     */
    public static void resize(String srcImgPath, String outImgPath,
                              int maxLength, float per) {
        // 得到图片
        BufferedImage src = InputImage(srcImgPath);
        int old_w = src.getWidth();
        // 得到源图宽
        int old_h = src.getHeight();
        // 得到源图长
        int new_w = 0;
        // 新图的宽
        int new_h = 0;

        if(old_h < maxLength && old_w < maxLength) {
            return;
        }

        // 新图的长
        BufferedImage tempImg = new BufferedImage(old_w, old_h,
                BufferedImage.TYPE_INT_RGB);
        Graphics2D g = tempImg.createGraphics();
        g.setColor(Color.white);
        // 从原图上取颜色绘制新图
        g.fillRect(0, 0, old_w, old_h);
        g.drawImage(src, 0, 0, old_w, old_h, Color.white, null);
        g.dispose();
        // 根据图片尺寸压缩比得到新图的尺寸
        if (old_w > old_h) {
            // 图片要缩放的比例
            new_w = maxLength;
            new_h = Math.round(old_h * ((float) maxLength / old_w));
        } else {
            new_w = Math.round(old_w * ((float) maxLength / old_h));
            new_h = maxLength;
        }
        BufferedImage newImg = new BufferedImage(new_w, new_h,
                BufferedImage.TYPE_INT_RGB);
        newImg.getGraphics().drawImage(
                tempImg.getScaledInstance(new_w, new_h, Image.SCALE_SMOOTH), 0,
                0, null);
        // 调用方法输出图片文件
        outImage(outImgPath, newImg, per);
    }

    /**
     * 将图片压缩成指定宽度， 高度等比例缩放
     *
     * @param srcImgPath
     * @param outImgPath
     * @param width
     * @param per
     */
    public static void resizeFixedWidth(String srcImgPath, String outImgPath,
                                        int width, float per) {
        // 得到图片
        BufferedImage src = InputImage(srcImgPath);
        int old_w = src.getWidth();
        // 得到源图宽
        int old_h = src.getHeight();
        // 得到源图长
        int new_w = 0;
        // 新图的宽
        int new_h = 0;
        // 新图的长
        BufferedImage tempImg = new BufferedImage(old_w, old_h,
                BufferedImage.TYPE_INT_RGB);
        Graphics2D g = tempImg.createGraphics();
        g.setColor(Color.white);
        // 从原图上取颜色绘制新图
        g.fillRect(0, 0, old_w, old_h);
        g.drawImage(src, 0, 0, old_w, old_h, Color.white, null);
        g.dispose();
        if (old_h < width && old_w < width) {
            new_h = old_h;
            new_w = old_w;
        }
        // 根据图片尺寸压缩比得到新图的尺寸
        else  {
            if (old_w > old_h) {
                // 图片要缩放的比例
                new_w = width;
                new_h = (int) Math.round(old_h * ((float) width / old_w));
            } else {
                new_w = (int) Math.round(old_w * ((float) width / old_h));
                new_h = width;
            }
        }
        BufferedImage newImg = new BufferedImage(new_w, new_h,
                BufferedImage.TYPE_INT_RGB);
        newImg.getGraphics().drawImage(
                tempImg.getScaledInstance(new_w, new_h, Image.SCALE_SMOOTH), 0,
                0, null);
        // 调用方法输出图片文件
        outImage(outImgPath, newImg, per);
    }

    /**
     * * 将图片文件输出到指定的路径，并可设定压缩质量
     *
     * @param outImgPath
     * @param newImg
     * @param per
     * @author cevencheng
     * */

    private static void outImage(String outImgPath, BufferedImage newImg, float per) {
        // 得到指定Format图片的writer
        Iterator iterator = ImageIO.getImageWritersByFormatName("jpg");
        ImageWriter writer = (ImageWriter) iterator.next();
        // 得到指定writer的输出参数设置(ImageWriteParam)
        ImageWriteParam iwp = writer.getDefaultWriteParam();
        iwp.setCompressionMode(ImageWriteParam.MODE_EXPLICIT); // 设置可否压缩
        iwp.setCompressionQuality(per); // 设置压缩质量参数
        iwp.setProgressiveMode(ImageWriteParam.MODE_DISABLED);
        ColorModel colorModel = ColorModel.getRGBdefault();
        // 指定压缩时使用的色彩模式
        iwp.setDestinationType(new javax.imageio.ImageTypeSpecifier(colorModel,
                colorModel.createCompatibleSampleModel(16, 16)));
        // 开始打包图片，写入byte[]
        ByteArrayOutputStream byteArrayOutputStream = new ByteArrayOutputStream(); // 取得内存输出流
        IIOImage image = new IIOImage(newImg, null, null);
        try {
            // 此处因为ImageWriter中用来接收write信息的output要求必须是ImageOutput
            // 通过ImageIo中的静态方法，得到byteArrayOutputStream的ImageOutput
            writer.setOutput(ImageIO
                    .createImageOutputStream(byteArrayOutputStream));
            writer.write(null, image, iwp);
            FileOutputStream outputStream  =new FileOutputStream(new File(outImgPath));
            outputStream.write(byteArrayOutputStream.toByteArray());
            outputStream.close();
        } catch (IOException e) {
            System.out.println("write errro");
            e.printStackTrace();
        }
    }

    public static long compressImage(String filePath) {
        File file = new File(filePath);
        long contentLength = file.length();
        //如果图片大于1200像素
        ImageUtil.resize(filePath, filePath, 1200);
        //如果图片大于500kb
        if (contentLength > 500 * 1024) {
            ImageUtil.resize(filePath, filePath, 1.0f, 0.5f);
        }
        //如果图片大于200kb
        else if (contentLength > 200 * 1024) {
            ImageUtil.resize(filePath, filePath, 1.0f, 0.6f);
        }
        //如果图片大于100kb
        else if (contentLength > 100 * 1024) {
            ImageUtil.resize(filePath, filePath, 1.0f, 0.7f);
        }
        //如果图片大于50kb
        else if (contentLength > 50 * 1024) {
            ImageUtil.resize(filePath, filePath, 1.0f, 0.8f);
        }
        return file.length();
    }

    /**
     * 自己设置压缩质量来把图片压缩成byte[]
     *
     * @param image
     *            压缩源图片
     * @param quality
     *            压缩质量，在0-1之间，
     * @return 返回的字节数组
     */
    private byte[] bufferedImageTobytes(BufferedImage image, float quality) {
        // 如果图片空，返回空
        if (image == null) {
            return null;
        }
        // 得到指定Format图片的writer
        Iterator iter = ImageIO
                .getImageWritersByFormatName("jpeg");// 得到迭代器
        ImageWriter writer = (ImageWriter) iter.next(); // 得到writer

        // 得到指定writer的输出参数设置(ImageWriteParam )
        ImageWriteParam iwp = writer.getDefaultWriteParam();
        iwp.setCompressionMode(ImageWriteParam.MODE_EXPLICIT); // 设置可否压缩
        iwp.setCompressionQuality(quality); // 设置压缩质量参数

        iwp.setProgressiveMode(ImageWriteParam.MODE_DISABLED);

        ColorModel colorModel = ColorModel.getRGBdefault();
        // 指定压缩时使用的色彩模式
        iwp.setDestinationType(new javax.imageio.ImageTypeSpecifier(colorModel,
                colorModel.createCompatibleSampleModel(16, 16)));

        // 开始打包图片，写入byte[]
        ByteArrayOutputStream byteArrayOutputStream = new ByteArrayOutputStream(); // 取得内存输出流
        IIOImage iIamge = new IIOImage(image, null, null);
        try {
            // 此处因为ImageWriter中用来接收write信息的output要求必须是ImageOutput
            // 通过ImageIo中的静态方法，得到byteArrayOutputStream的ImageOutput
            writer.setOutput(ImageIO
                    .createImageOutputStream(byteArrayOutputStream));
            writer.write(null, iIamge, iwp);
        } catch (IOException e) {
            System.out.println("write errro");
            e.printStackTrace();
        }
        return byteArrayOutputStream.toByteArray();
    }

    /**
     * 图片剪切工具方法
     *
     * @param srcfile 源图片
     * @param outfile 剪切之后的图片
     * @param x 剪切顶点 X 坐标
     * @param y 剪切顶点 Y 坐标
     * @param width 剪切区域宽度
     * @param height 剪切区域高度
     *
     * @throws IOException
     * @author cevencheng
     */
    public static void cut(File srcfile, File outfile, int x, int y, int width, int height) throws IOException {
        FileInputStream is = null;
        ImageInputStream iis = null;
        try {
            // 读取图片文件
            is = new FileInputStream(srcfile);

            /*
             * 返回包含所有当前已注册 ImageReader 的 Iterator，这些 ImageReader 声称能够解码指定格式。
             * 参数：formatName - 包含非正式格式名称 .（例如 "jpeg" 或 "tiff"）等 。
             */
            Iterator<ImageReader> it = ImageIO.getImageReadersByFormatName("jpg");
            ImageReader reader = it.next();
            // 获取图片流
            iis = ImageIO.createImageInputStream(is);

            /*
             * <p>iis:读取源.true:只向前搜索 </p>.将它标记为 ‘只向前搜索’。
             * 此设置意味着包含在输入源中的图像将只按顺序读取，可能允许 reader 避免缓存包含与以前已经读取的图像关联的数据的那些输入部分。
             */
            reader.setInput(iis, true);

            /*
             * <p>描述如何对流进行解码的类<p>.用于指定如何在输入时从 Java Image I/O
             * 框架的上下文中的流转换一幅图像或一组图像。用于特定图像格式的插件 将从其 ImageReader 实现的
             * getDefaultReadParam 方法中返回 ImageReadParam 的实例。
             */
            ImageReadParam param = reader.getDefaultReadParam();

            /*
             * 图片裁剪区域。Rectangle 指定了坐标空间中的一个区域，通过 Rectangle 对象
             * 的左上顶点的坐标（x，y）、宽度和高度可以定义这个区域。
             */
            Rectangle rect = new Rectangle(x, y, width, height);

            // 提供一个 BufferedImage，将其用作解码像素数据的目标。
            param.setSourceRegion(rect);

            /*
             * 使用所提供的 ImageReadParam 读取通过索引 imageIndex 指定的对象，并将 它作为一个完整的
             * BufferedImage 返回。
             */
            BufferedImage bi = reader.read(0, param);

            // 保存新图片
            ImageIO.write(bi, "jpg", outfile);
        } finally {
            if (is != null) {
                is.close();
            }
            if (iis != null) {
                iis.close();
            }
        }
    }

    public static void main(String args[]) throws Exception {
        String srcImg = "C:\\Users\\Riko\\Desktop\\test.jpg";
        String tarDir;
        File srcfile = new File(srcImg);
        tarDir = srcfile.getParent();
        File file = new File(srcImg);
        System.out.println(file.length());
        resize(srcImg, srcImg, 1200);
        System.out.println(file.length());
        resize(srcImg, srcImg, 1.0f, 0.9f);
        System.out.println(file.length());
        resize(srcImg, srcImg, 1.0f, 0.8f);
        System.out.println(file.length());
        resize(srcImg, srcImg, 1.0f, 0.7f);
        System.out.println(file.length());
        resize(srcImg, srcImg, 1.0f, 0.6f);
        System.out.println(file.length());
        resize(srcImg, srcImg, 1.0f, 0.5f);
        System.out.println(file.length());
    }
}